package Algorithm.kmp;

import java.util.Arrays;

/**
 * KMP算法
 */
public class KMPAlgorithm {
    public static void main(String[] args) {
        String s1 = "BBC ABCDAB ABCDABCDABDE";
        String s2 = "ABCDABD";

        int[] ints = kmpNext(s2);
        System.out.println(Arrays.toString(ints));
        int i = kmpSearch(s1, s2, ints);
        System.out.println(i);
    }

    //写出kmp搜索算法

    /**
     *
     * @param str1  源字符串
     * @param str2  待查找的串
     * @param next  部分匹配表（字串对应的部分匹配表）
     * @return  匹配到的第一个位置
     */
    public static int kmpSearch(String str1,String str2,int[] next){
        //遍历str1
        for (int i = 0, j = 0; i < str1.length(); i++) {

            //当不匹配的时候，会直接调整j的大小
            //kmp算法核心
            while (j > 0 && str1.charAt(i) != str2.charAt(j)) {
                j = next[j - 1];
            }

            if (str1.charAt(i) == str2.charAt(j)) {
                j++;
            }
            if (j == str2.length()) {
                //找到了
                return i - j + 1;
            }
        }
        return -1;
    }

    //获取到子串的部分匹配值
    public static int[] kmpNext(String dest){
        //创建一个数组保存部分匹配值
        int[] next = new int[dest.length()];
        next[0] = 0; //如果字符串长度为1，则部分匹配值一定为0
        for (int i = 1,j=0; i <dest.length() ; i++) {
            //当dest.charAt(i) ！= dest.charAt(j)
            //需要从next[j-1]获取新的j
            while (j > 0 && dest.charAt(i) != dest.charAt(j)){
                j = next[j-1];
            }

            if(dest.charAt(i) == dest.charAt(j)){
                //部分匹配值就+1
                j++;
            }
            next[i] = j;
        }
        return next;
    }
}
